# Generation
import os
from dotenv import load_dotenv
from openai import OpenAI


def generate(query: str, chunks: list[str]) -> str:

    load_dotenv()
    client = OpenAI(
        api_key=os.getenv("NG_API_KEY"),
        base_url="https://ng.115.zone/v1/",
    )
    joined_chunks = "\n\n".join(chunks)
    prompt = f"""
    你是一位专业的知识助手，名字叫XDUMsgBot，你可以根据相关片段中的信息回答用户关于西安电子科技大学的问题。
    请根据用户的问题和下列片段生成准确的回应。

    用户问题：{query}
    
    相关片段：
    {joined_chunks}
    
    请基于上述内容作答，如果没有明确的信息可供参考，请回答不知道，不要编造信息。
    
    """
    print(prompt)
    response = client.chat.completions.create(
        model="deepseek-v3",
        messages=[{"role": "user", "content": prompt}],
        temperature=0.7,
        max_tokens=5000,
    )
    # 提取并打印生成的回复
    generated_text = response.choices[0].message.content
    return generated_text